import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
from sklearn.datasets import load_iris

def plot_iris_data(data,col1,col2):
    """
    :param data: 数据集
    :param col1: 特征1
    :param col2: 特征2
    """
    # 绘制数据集的分布图
    # 1.绘制数据集的分布图
    sns.lmplot(data = data,x=col1,y=col2,
               hue='target',fit_reg= False)
    plt.title('鸢尾花数据展示')
    plt.xlabel(col1)
    plt.ylabel(col2)
    plt.show()

if __name__ == '__main__':
    # 加载鸢尾花数据集
    iris = load_iris()
    # 数据可视化
    # 1.数据类型的转换，用DataFrame存储数据
    iris_data = pd.DataFrame(data=iris.data, columns=iris.feature_names)
    iris_data['target'] = iris.target
    print(iris_data)

    # 2.绘制数据集的分布图
    plot_iris_data(iris_data,'sepal length (cm)','petal width (cm)')
